#include <stdio.h>

// 辅助函数：调整堆
void heapify(int arr[], int n, int i) {
    // int largest = i; // 初始化最大值为根节点
    // int left = 2 * i + 1; // 左子节点
    // int right = 2 * i + 2; // 右子节点

    // // 如果左子节点存在且大于根节点
    // if (left < n && arr[left] > arr[largest])
    //     largest = left;

    // // 如果右子节点存在且大于最大值
    // if (right < n && arr[right] > arr[largest])
    //     largest = right;

    // // 如果最大值不是根节点
    //大根堆，将根节点设置为较大值
    // if (largest != i) {
    //     int temp = arr[i];
    //     arr[i] = arr[largest];
    //     arr[largest] = temp;

    //     // 递归地调整受影响的子堆
    //     heapify(arr, n, largest);
    // }
    int left = i*2 + 1;
    int right = i*2 + 2;
    int largest = i;

    if(left < n && arr[largest] < arr[left])
    {
        largest = left;
    }
    if(right < n && arr[largest] < arr[right])
    {
        largest = right;
    }
    if(largest != i)
    {
        int temp = arr[largest];
        arr[largest] = arr[i];
        arr[i] = temp;

        heapify(arr,n,largest);
    }
}

// 主函数：堆排序
void heapSort(int arr[], int n) {
    // 构建堆
    for (int i = n / 2 - 1; i >= 0; i--)//从最后一个非叶子节点开始调整或建立（实际上是在一个数组中调整下标）
        heapify(arr, n, i);

    // 一个接一个地从堆中取出元素
    // for (int i = n - 1; i > 0; i--) {
    //     // 将当前根节点与数组的最后一个元素交换
    //     int temp = arr[0];
    //     arr[0] = arr[i];
    //     arr[i] = temp;

    //     // 调整堆
    //     heapify(arr, i, 0);
    // }

    for( int i = n - 1; i > 0; i--)
    {
        int temp = arr[0];
        arr[0] = arr[i];
        arr[i] = temp;

        heapify(arr,i,0); //如果这里写为n的话，实际上什么也不做，因为0位置实际上就是最大值
    }
}

// 辅助函数：打印数组
void printArray(int arr[], int n) {
    for (int i = 0; i < n; i++)
        printf("%d ", arr[i]);
    printf("\n");
}

int main() {
    int arr[] = {40, 190, 3, 5, 1};
    int n = sizeof(arr) / sizeof(arr[0]);

    heapSort(arr, n);

    printf("Sorted array is: ");
    printArray(arr, n);

    return 0;
}
